# 给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。
# 你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
#
# 你可以按任意顺序返回答案。
#
#  示例 1：
# 输入：nums = [2,7,11,15], target = 9
# 输出：[0,1]
# 解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。
#
#  示例 2：
# 输入：nums = [3,2,4], target = 6
# 输出：[1,2]
#
#  示例 3：
# 输入：nums = [3,3], target = 6
# 输出：[0,1]
from typing import List


class Solution:
    """
        哈希表解法
        key 为 nums[i] value 为 i
    """

    def twoSum2(self, nums: List[int], target: int) -> List[int]:
        hashmap = {}
        for i, num in enumerate(nums):
            if target - num in hashmap:
                return [hashmap[target - num], i]
            hashmap[num] = i
        return []

    """
    先排序（带上下标），然后转化成leetcode1 使用双指针解法即可
    """

    def twoSum1(self, nums: List[int], target: int) -> List[int]:
        newNums = sorted([(num, i) for i, num in enumerate(nums)], key=lambda x: x[0])
        i, j = 0, len(newNums) - 1
        while i < j:
            if newNums[i][0] + newNums[j][0] == target:
                return [newNums[i][1], newNums[j][1]]
            elif newNums[i][0] + newNums[j][0] < target:
                i += 1
            else:
                j -= 1
        return []

    def twoSum(self, nums: List[int], target: int) -> List[int]:
        return self.twoSum1(nums, target)


if __name__ == "__main__":
    nums = [2, 7, 11, 15]
    target = 9
    # nums = [3, 2, 4]
    # target = 6

    print(Solution().twoSum(nums, target))
